#pragma once

#include <iostream>
#include <string>
#include <unistd.h>
#include <sys/wait.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>

#include "../comm/util.hpp"  // 引入公共的方法
#include "../comm/log.hpp"   // 引入日志功能

// 该模块只负责代码的编译

namespace ns_compiler // 带上命名空间, 方便管理以及避免命名冲突
{
    using namespace ns_util; // 引入公共的方法
    using namespace ns_log;  // 引入日志功能

    class Compiler
    {
    public:
        Compiler()
        {}

        ~Compiler()
        {}

        // file_name: 输入参数,需要编译代码的文件名(不带.cpp后缀)
        // file_name: 1234
        // 1234 -> ./temp/1234.cpp (源文件)
        // 1234 -> ./temp/1234.exe (可执行程序)
        // 1234 -> ./temp/1234.stderr (标准错误文件)
        // ./temp 是存放临时文件的文件夹
        // Compile 函数的功能是对存在的源文件进行编译
        // 返回值: 编译成功, 返回 true; 编译失败, 返回 false
        static bool Compile(const std::string& file_name)
        {
            pid_t id = fork(); // 创建子进程来负责编译
            if(id < 0)
            {
                LOG(ERROR) << "编译时创建子进程失败" << "\n";
                return false;  // 创建子进程失败, 直接返回 false
            }
            else if(id == 0) // 子进程
            {
                umask(0);
                // 创建存储编译错误信息的文件
                int _stderr = open(PathUtil::CompilerError(file_name).c_str(), O_CREAT | O_WRONLY, 0644);  // 创建标准错误文件
                if(_stderr < 0)
                {
                    LOG(WARNING) << "创建 stderr 文件失败" << "\n";
                    exit(1); // 创建文件失败, 直接退出子进程
                }
                // 将编译错误信息重定向到 _stderr 中
                dup2(_stderr, 2); 

                // 程序替换并不会影响进程的文件描述符表
                // 子进程: 调用编译器完成对代码的编译工作
                // l: 可变参数列表
                // p: 不需要执行可执行程序所在的路径,只需要指明可执行程序的名字即可
                // g++ -o target src -std=c++11
                execlp("g++", "g++", "-o", PathUtil::Exe(file_name).c_str(),\
                PathUtil::Src(file_name).c_str(), "-std=c++11", "-D", "CompileOnline", nullptr/*以 nullptr 表示参数的结束*/);
                LOG(ERROR) << "启动编译器失败, 传入的参数可能有误" << "\n";
                exit(2); // 进程替换失败
            }
            else // 父进程
            {
                // 父进程阻塞等待子进程退出
                waitpid(id, nullptr, 0);
                // 编译是否成功可以通过是否形成了可执行程序来判断
                if(FileUtil::IsFileExists(PathUtil::Exe(file_name)))
                {
                    LOG(INFO) << PathUtil::Src(file_name) << " 文件编译成功" << "\n";
                    return true;
                }
            }

            LOG(ERROR) << "编译失败, 没有形成可执行程序" << "\n";
            return false; // 编译失败
        }
    };
}